tensor([[0.5200, 0.5200, 0.5200, 0.5200, 0.5200],
        [0.4800, 0.4800, 0.4800, 0.4800, 0.4800]], grad_fn=<RoundBackward1>)